package com.utn.disenio.entrega3;

import java.util.ArrayList;

public class FiltroDeUbicacion implements TipoDeFiltro{
	private String ubicacion;
	
	public FiltroDeUbicacion(String ubicacion){
		this.ubicacion = ubicacion;
	}
	
	public ArrayList<Viaje> filtrar(ArrayList<Viaje> viajesAFiltrar) {
		ArrayList<Viaje> listaFiltrada = new ArrayList<Viaje>();
		for(Viaje unViaje: viajesAFiltrar)
		{
			if(unViaje.getAsientoUno().getUbicacion().equals(ubicacion) &&
			  (unViaje.getAsientoDos() != null && unViaje.getAsientoDos().getUbicacion().equals(ubicacion)) &&
			  (unViaje.getAsientoTres() != null && unViaje.getAsientoTres().getUbicacion().equals(ubicacion))
			)
				listaFiltrada.add(unViaje);

		}
		return listaFiltrada;

	}

}
